package fun.carliels.doraemon.doc.excel;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.function.BiFunction;
import java.util.logging.Logger;

/**
 * Description: 读取Excel内容
 */
public class ExcelReader {

    /**
     * 日志打印类
     */
    private static Logger logger = Logger.getLogger(ExcelReader.class.getName());

    private static final String XLS = "xls";
    private static final String XLSX = "xlsx";

    /**
     *  读取Excel, 默认首行为标题行
     * @param fileName 文件url
     * @param function 转换函数BiFunction<List<String> cols, List<String> titles, R>
     * @param <R> 读取结果
     * @return
     */
    public static<R> List<R> readExcel(String fileName,BiFunction<List<String>, List<String>, R> function) {
        return readExcel(fileName, function, 0);
    }

    /**
     * 读取Excel文件内容
     * @param fileName 要读取的Excel文件所在路径
     * @param function 转换函数BiFunction<List<String> cols, List<String> titles, R>
     * @param titleRowNum 标题行号,默认为首行0
     * @return 读取结果列表，读取失败时返回null
     */
    public static<R> List<R> readExcel(String fileName, BiFunction<List<String>, List<String>, R> function, int titleRowNum) {

        Workbook workbook = null;
        FileInputStream inputStream = null;

        try {
            // 获取Excel后缀名
            String fileType = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length());
            // 获取Excel文件
            File excelFile = new File(fileName);
            if (!excelFile.exists()) {
                logger.warning("指定的Excel文件不存在！");
                return null;
            }

            // 获取Excel工作簿
            inputStream = new FileInputStream(excelFile);
            workbook = getWorkbook(inputStream, fileType);

            // 读取excel中的数据
            List<R> resultDataList = parseExcel(workbook, function, titleRowNum);

            return resultDataList;
        } catch (Exception e) {
            logger.warning("解析Excel失败，文件名：" + fileName + " 错误信息：" + e.getMessage());
            return null;
        } finally {
            try {
                if (null != workbook) {
                    workbook.close();
                }
                if (null != inputStream) {
                    inputStream.close();
                }
            } catch (Exception e) {
                logger.warning("关闭数据流出错！错误信息：" + e.getMessage());
                return null;
            }
        }
    }

    /**
     * 根据文件后缀名类型获取对应的工作簿对象
     * @param inputStream 读取文件的输入流
     * @param fileType 文件后缀名类型（xls或xlsx）
     * @return 包含文件数据的工作簿对象
     * @throws IOException
     */
    private static Workbook getWorkbook(InputStream inputStream, String fileType) throws IOException {
        Workbook workbook = null;
        if (fileType.equalsIgnoreCase(XLS)) {
            workbook = new HSSFWorkbook(inputStream);
        } else if (fileType.equalsIgnoreCase(XLSX)) {
            workbook = new XSSFWorkbook(inputStream);
        }
        return workbook;
    }

    /**
     * 解析Excel数据
     * @param workbook Excel工作簿对象
     * @return 解析结果
     */
    private static<R> List<R> parseExcel(Workbook workbook, BiFunction<List<String>, List<String>, R> function, int titleRowNum) {
        List<R> resultDataList = new ArrayList<>();
        // 解析sheet
        for (int sheetNum = 0; sheetNum < workbook.getNumberOfSheets(); sheetNum++) {
            Sheet sheet = workbook.getSheetAt(sheetNum);

            // 校验sheet是否合法
            if (sheet == null) {
                continue;
            }

            // 获取第一行数据，判定excel是否为空
            int firstRowNum = sheet.getFirstRowNum();
            Row titleRow = sheet.getRow(firstRowNum);
            if (null == titleRow) {
                logger.warning("解析Excel失败，在第一行没有读取到任何数据！");
            }

            // 处理标题行
            if (titleRowNum > 0) {
                firstRowNum = titleRowNum;
                titleRow = sheet.getRow(titleRowNum);
            }

            // 解析标题行
            List<String> titles = readRowToList(titleRow);

            // 解析每一行的数据，构造数据对象
            int rowStart = firstRowNum + 1;
            int rowEnd = sheet.getPhysicalNumberOfRows();
            for (int rowNum = rowStart; rowNum < rowEnd; rowNum++) {
                Row row = sheet.getRow(rowNum);

                if (null == row) {
                    continue;
                }

                List<String> cols = readRowToList(row);

                R resultData = function.apply(cols, titles);
                if (null == resultData) {
                    logger.warning("第 " + row.getRowNum() + "行数据不合法，已忽略！");
                    continue;
                }
                resultDataList.add(resultData);
            }
        }

        return resultDataList;
    }

    private static List<String> readRowToList(Row row) {
        List<String> result = new ArrayList<>();
        int firstCellNum = row.getFirstCellNum();
        int lastCellNum = row.getLastCellNum();
        for (int i=firstCellNum; i<lastCellNum; i++) {
            result.add(convertCellValueToString(row.getCell(i)));
        }

        return result;
    }

    /**
     * 将单元格内容转换为字符串
     * @param cell
     * @return
     */
    public static String convertCellValueToString(Cell cell) {
        if(cell==null){
            return null;
        }
        return new DataFormatter().formatCellValue(cell);
//        String returnValue = null;
//        switch (cell.getCellType()) {
//            case NUMERIC:   //数字
//                Double doubleValue = cell.getNumericCellValue();
//                // 格式化科学计数法，取一位整数
//                DecimalFormat df = new DecimalFormat("0");
//                returnValue = df.format(doubleValue);
//                break;
//            case STRING:    //字符串
//                returnValue = cell.getStringCellValue();
//                break;
//            case BOOLEAN:   //布尔
//                Boolean booleanValue = cell.getBooleanCellValue();
//                returnValue = booleanValue.toString();
//                break;
//            case BLANK:     // 空值
//                break;
//            case FORMULA:   // 公式
//                returnValue = cell.getCellFormula();
//                break;
//            case ERROR:     // 故障
//                break;
//            default:
//                break;
//        }
//        return returnValue;
    }

    /**
     * 安全取值
     * @param cols
     * @param index
     * @return
     */
    public static String getColValue(List<String> cols, int index) {
        if (index > -1 && index < cols.size()) {
            return cols.get(index);
        }
        return null;
    }

    /**
     * 安全取值-带默认值
     * @param cols
     * @param index
     * @param defaultValue
     * @return
     */
    public static String getColValueDefault(List<String> cols, int index, String defaultValue) {
        if (index > -1 && index < cols.size()) {
            return cols.get(index);
        }
        return defaultValue;
    }
}

